package main

import (
	"context"
	"fmt"
	"log"
	"net/http"
	"os"
	"os/signal"
	"time"

	"github.com/gin-gonic/gin"
	//"github.com/spf13/viper"
	"go.uber.org/zap"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"

	// 添加这些导入
	"go-admin-system/config"
	"go-admin-system/database"
	"go-admin-system/middleware"
	"go-admin-system/routes"
)

func main() {
	// 加载配置
	cfg := config.GetConfig()

	// 初始化数据库连接
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
		cfg.Database.Username,
		cfg.Database.Password,
		cfg.Database.Host,
		cfg.Database.Port,
		cfg.Database.Name)

	db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		log.Fatal("数据库连接失败:", err)
	}

	// 自动迁移模型
	database.MigrateModels(db)

	// 创建Gin引擎
	r := gin.Default()

	// 初始化zap日志
	logger, _ := zap.NewProduction()
	defer logger.Sync()

	// 注册中间件（使用 middleware 包中的函数）
	middleware.SetupMiddleware(r, db, logger, cfg)

	// 注册路由
	routes.SetupRoutes(r, db, logger)

	// 启动服务器
	addr := ":" + cfg.Port
	srv := &http.Server{
		Addr:    addr,
		Handler: r,
	}

	go func() {
		if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
			log.Fatalf("服务器启动失败: %v", err)
		}
	}()

	fmt.Printf("服务器已启动，监听端口:%s\n", cfg.Port)

	// 优雅关闭
	quit := make(chan os.Signal, 1)
	signal.Notify(quit, os.Interrupt)
	<-quit
	fmt.Println("接收到关闭信号，正在优雅关闭服务器...")

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	if err := srv.Shutdown(ctx); err != nil {
		log.Fatal("服务器强制关闭:", err)
	}
	fmt.Println("服务器已优雅关闭")
}
